import os
import sys

# 将项目根目录添加到Python模块搜索路径
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from utils.logger_config import logger

# 定义图片所在目录
image_dir = "../static/images/src-face/zhangjingyi"

# 定义支持的图片扩展名
image_extensions = (".jpg", ".jpeg", ".png", ".bmp", ".webp")

# 获取目录下所有图片文件
image_files = [f for f in os.listdir(image_dir) if f.lower().endswith(image_extensions)]

# 按文件名排序
image_files.sort()

# 遍历图片文件并重命名
for index, old_name in enumerate(image_files, start=1):
    # 获取文件扩展名
    file_extension = os.path.splitext(old_name)[1]
    # 若数字小于 10，添加前缀 0
    if index < 10:
        new_name = f"zhangjingyi-0{index}{file_extension}"
    else:
        new_name = f"zhangjingyi-{index}{file_extension}"
    # 构建旧文件的完整路径
    old_path = os.path.join(image_dir, old_name)
    # 构建新文件的完整路径
    new_path = os.path.join(image_dir, new_name)

    # 检查目标文件是否存在，如果存在则删除
    if os.path.exists(new_path):
        os.remove(new_path)
        logger.info(f"已删除已存在的文件: {new_path}")

    # 重命名文件
    os.rename(old_path, new_path)
    logger.info(f"已将 {old_name} 重命名为 {new_name}")
